package cuit.demo.controller;

import cuit.demo.Entity.User;
import cuit.demo.service.MyService;
import jakarta.servlet.http.HttpSession;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.stereotype.Controller;
import org.springframework.ui.Model;
import org.springframework.web.bind.annotation.GetMapping;
import org.springframework.web.bind.annotation.PostMapping;
import org.springframework.web.bind.annotation.RequestParam;

@Controller
public class LoginController {

    @Autowired
    private MyService myService;

    // 显示登录页面
    @GetMapping("/login")
    public String showLoginPage() {
        return "login";
    }

    // 处理用户登录
    @PostMapping("/login")
    public String loginUser(
            @RequestParam int userid,
            @RequestParam String password,
            Model model, HttpSession session) {

        // 查找用户
        User user = myService.findUserById(userid);

        // 验证用户
        if (user != null && user.getUserpwd().equals(password)) { // 这里建议用密码加密方法来比
            // 登录成功，可能在这里保存用户会话或添加成功消息
            model.addAttribute("message", "登录成功！欢迎，" + user.getUsername());
            session.setAttribute("user", user);
            return "index"; // 登录成功后的主页，您可以替换为您自己的主页
        } else {
            // 登录失败，返回登录页面并显示错误信息
            model.addAttribute("error", "用户名或密码错误，请重试！");
            return "login"; // 返回登录页面
        }
    }
}